Node / ReticulumProjects / Sideband.git / files / sbapp / plyer / tests / test_uniqueid.py

Displaying Raw • Download

T8b949e'''

TestUniqueID

============

Tested platforms:

* Windows

'''

Tff7b72import T7ee787unittest

Tff7b72from T7ee787mock Tff7b72import Te6edf3patchTb4b4b4, Te6edf3Mock

Tff7b72from T7ee787plyerT7ee787.T7ee787testsT7ee787.T7ee787common Tff7b72import Te6edf3PlatformTestTb4b4b4, Te6edf3platform_import

Tff7b72class T56d364TestUniqueIDTb4b4b4(Te6edf3unittestTff7b72.Td2a8ffTestCaseTb4b4b4)Tb4b4b4:

T8b949e'''

TestCase for plyer.uniqueid.

'''

Tff7b72def Td2a8fftest_uniqueidTb4b4b4(Tff7b72selfTb4b4b4)Tb4b4b4:

T8b949e'''

General all platform test for plyer.uniqueid.

'''

Tff7b72from T7ee787plyer Tff7b72import Te6edf3uniqueid

Tff7b72selfTff7b72.Td2a8ffassertTrueTb4b4b4(Tffa657lenTb4b4b4(Te6edf3uniqueidTff7b72.Td2a8ffidTb4b4b4) Tff7b72> T79c0ff0Tb4b4b4)

Tf0883e@PlatformTestTb4b4b4(Ta5d6ff'Ta5d6ffwinTa5d6ff'Tb4b4b4)

Tff7b72def Td2a8fftest_uniqueid_winTb4b4b4(Tff7b72selfTb4b4b4)Tb4b4b4:

T8b949e'''

Test Windows API for plyer.uniqueid.

'''

Tff7b72tryTb4b4b4:

Tff7b72from T7ee787winreg Tff7b72import Tb4b4b4(

Te6edf3HKEY_LOCAL_MACHINE Tff7b72as Te6edf3HKLMTb4b4b4,

Te6edf3KEY_READ Tff7b72as Te6edf3READTb4b4b4, Te6edf3KEY_WOW64_64KEY Tff7b72as Te6edf3VIEW

Tb4b4b4)

Tff7b72except Tf85149ImportErrorTb4b4b4:

Tff7b72from T7ee787_winreg Tff7b72import Tb4b4b4(

Te6edf3HKEY_LOCAL_MACHINE Tff7b72as Te6edf3HKLMTb4b4b4,

Te6edf3KEY_READ Tff7b72as Te6edf3READTb4b4b4, Te6edf3KEY_WOW64_64KEY Tff7b72as Te6edf3VIEW

Tb4b4b4)

T8b949e# mock the 'regedit' alias for winreg,

T8b949e# see if the import passes and get the instance

Te6edf3regedit_mod Tff7b72= Ta5d6ff'Ta5d6ffplyer.platforms.win.uniqueid.regeditTa5d6ff'

Tff7b72with Te6edf3patchTb4b4b4(Te6edf3targetTff7b72=Te6edf3regedit_modTb4b4b4)Tb4b4b4:

Te6edf3uniqueid_ Tff7b72= Te6edf3platform_importTb4b4b4(

Te6edf3platformTff7b72=Ta5d6ff'Ta5d6ffwinTa5d6ff'Tb4b4b4,

Te6edf3module_nameTff7b72=Ta5d6ff'Ta5d6ffuniqueidTa5d6ff'

Tb4b4b4)

Te6edf3uniqueid Tff7b72= Te6edf3uniqueid_Tff7b72.Td2a8ffinstanceTb4b4b4(Tb4b4b4)

Tff7b72selfTff7b72.Td2a8ffassertIsInstanceTb4b4b4(Te6edf3uniqueid_Tff7b72.Td2a8ffregeditTb4b4b4, Te6edf3MockTb4b4b4)

T8b949e# out of mocking block, regedit should be a winreg module

Tff7b72selfTff7b72.Td2a8ffassertIsInstanceTb4b4b4(Te6edf3uniqueid_Tff7b72.Td2a8ffregeditTb4b4b4, Tffa657typeTb4b4b4(Te6edf3unittestTb4b4b4)Tb4b4b4)

T8b949e# OpenKey is supposed to return a handle to registry key

Te6edf3regedit_opkey Tff7b72= Ta5d6ff'Ta5d6ffplyer.platforms.win.uniqueid.regedit.OpenKeyTa5d6ff'

Tff7b72with Te6edf3patchTb4b4b4(Te6edf3targetTff7b72=Te6edf3regedit_opkeyTb4b4b4, Te6edf3return_valueTff7b72=Ta5d6ff'Ta5d6ffunicornTa5d6ff'Tb4b4b4) Tff7b72as Te6edf3opkeyTb4b4b4:

T8b949e# QueryValueEx is supposed to return 2 packed values

T8b949e# (key, type_id)

Te6edf3queryval Tff7b72= Ta5d6ff'Ta5d6ffplyer.platforms.win.uniqueid.regedit.QueryValueExTa5d6ff'

Te6edf3retval Tff7b72= Tb4b4b4(Ta5d6ff'Ta5d6ffuniqueTa5d6ff'Tb4b4b4, Tff7b72NoneTb4b4b4)

Tff7b72with Te6edf3patchTb4b4b4(Te6edf3targetTff7b72=Te6edf3queryvalTb4b4b4, Te6edf3return_valueTff7b72=Te6edf3retvalTb4b4b4) Tff7b72as Te6edf3queryTb4b4b4:

Te6edf3uid Tff7b72= Te6edf3uniqueidTff7b72.Td2a8ffid

Te6edf3opkeyTff7b72.Td2a8ffassert_called_once_withTb4b4b4(

T8b949e# key, subkey

Te6edf3HKLMTb4b4b4, Ta5d6ffrTa5d6ff'Ta5d6ffSOFTWARETffea00\\Ta5d6ffMicrosoftTffea00\\Ta5d6ffCryptographyTa5d6ff'Tb4b4b4,

T8b949e# reserved integer (has to be 0 - zero), access mask

T79c0ff0Tb4b4b4, Te6edf3READ Tff7b72| Te6edf3VIEW

Tb4b4b4)

Te6edf3queryTff7b72.Td2a8ffassert_called_once_withTb4b4b4(Ta5d6ff'Ta5d6ffunicornTa5d6ff'Tb4b4b4, Ta5d6ff'Ta5d6ffMachineGuidTa5d6ff'Tb4b4b4)

Tff7b72selfTff7b72.Td2a8ffassertEqualTb4b4b4(Te6edf3uidTb4b4b4, Te6edf3retvalTb4b4b4[T79c0ff0Tb4b4b4]Tb4b4b4)

Tff7b72if Tff7b72__name__ Tff7b72== Ta5d6ff'Ta5d6ff__main__Ta5d6ff'Tb4b4b4:

Te6edf3unittestTff7b72.Td2a8ffmainTb4b4b4(Tb4b4b4)